TypeError: list indices must be integers or slices not list (Python)